import os
import pandas as pd
import glob 

def create_test_datasets(testlabels='input/test/test_labels.csv'): # change the 
    # read test_labels.csv
    test=pd.read_csv(testlabels)
    #change columns
    columns=['image_path','dataset','scene','rotation_matrix','translation_vector']

    test=test.loc[:,columns]
    test.to_csv("input/test/test_labels.csv")
    print(test)
    # 
    return test

create_test_datasets()